import tinycudann as tcnn
import torch

# 定义哈希编码器配置
config = {
    "otype": "HashGrid",
    "n_levels": 16,
    "n_features_per_level": 2,
    "log2_hashmap_size": 19,
    "base_resolution": 16,
    "per_level_scale": 1.447269237440378
}

# 创建编码器并测试前向传播
encoder = tcnn.Encoding(3, config)  # 3 表示输入维度（x,y,z）
input = torch.randn(1, 3, device="cuda")  # 生成随机输入（需在 CUDA 上运行）
output = encoder(input)
print("输出形状:", output.shape)  # 应输出 (1, 32)（16层×2特征/层）